from transformers import BertModel
import datasets
import torch
import torch.nn as nn

EVAL_BATCH = 8
EVAL_EPOCH = 1
CRITERION = nn.CrossEntropyLoss()

# repair REPOCARD_FILENAME
if not hasattr(datasets.config, 'REPOCARD_FILENAME'):
    datasets.config.REPOCARD_FILENAME = 'README.md'
from modelscope.msdatasets import MsDataset


a=torch.load("./model_state_checkpoint5.pth")


print(a)
